Skip to content

Conversation

@gongchensu
Copy link
Collaborator

@gongchensu gongchensu commented Nov 7, 2025

  1. 基于 @spike-zhu 的添加Module的PR,开发ModuleList和Parameter,并增加了两个的测试
  2. InfiniCoreModule中也相应做了些改动
image image image

@gongchensu gongchensu self-assigned this Nov 7, 2025
@gongchensu gongchensu linked an issue Nov 7, 2025 that may be closed by this pull request
@gongchensu gongchensu force-pushed the feature/add_nn_interface branch from 4f473b5 to 4e35d6a Compare November 7, 2025 11:14
@gongchensu gongchensu force-pushed the feature/add_nn_interface branch from 75d34fa to cf2811f Compare November 14, 2025 03:24
raise KeyError("not support")


Module = InfiniCoreModule
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

最后的 Module = InfiniCoreModule,是不是可以在 init.py 中使用 from .module import InfiniCoreModule as Module 替代

@gongchensu gongchensu force-pushed the feature/add_nn_interface branch 2 times, most recently from 5c52d62 to e759de1 Compare November 17, 2025 03:25
@gongchensu gongchensu force-pushed the feature/add_nn_interface branch from e759de1 to ee722eb Compare November 17, 2025 03:27
@pengcheng888 pengcheng888 merged commit 3567f92 into InfiniTensor:main Nov 17, 2025
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[DEV] infiniCore的nn对齐ModuleList和Parameter

4 participants